package com.uxhao.oss.service.impl;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.uxhao.oss.service.OssService;
import com.uxhao.oss.utils.ConstantPropertiesUtils;
import org.joda.time.DateTime;
import org.springframework.stereotype.Service;
import org.springframework.util.StringUtils;
import org.springframework.web.multipart.MultipartFile;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.InputStream;
import java.util.UUID;

/**
 * @author uxhao
 * @date 2021/1/9 21:17
 */
@Service
public class OssServiceImpl implements OssService {
    // 图片类型
    private static String TYPESTR[] = {".png", ".jpg", ".bmp", ".gif", ".jpeg"};

    //上传代码到Oss
    @Override
    public String uploadFileAvatar(MultipartFile file) {
        // yourEndpoint填写Bucket所在地域对应的Endpoint。以华东1（杭州）为例，Endpoint填写为https://oss-cn-hangzhou.aliyuncs.com。
        String endpoint = ConstantPropertiesUtils.END_POINT;
        // 阿里云账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM用户进行API访问或日常运维，请登录RAM控制台创建RAM用户。
        String accessKeyId = ConstantPropertiesUtils.ACCESS_KEY_ID;
        String accessKeySecret = ConstantPropertiesUtils.ACCESS_KEY_SECRET;
        String bucketName = ConstantPropertiesUtils.BUCKET_NAME;

        // 填写本地文件的完整路径。如果未指定本地路径，则默认从示例程序所属项目对应本地路径中上传文件流。
        InputStream inputStream = null;
        OSS ossClient = new OSSClientBuilder().build(
                endpoint,
                accessKeyId,
                accessKeySecret);
        try {

            boolean flag = false;
            // 判断 文件格式
            for (String type : TYPESTR) {
                if (StringUtils.endsWithIgnoreCase(file.getOriginalFilename(), type)) {
                    flag = true;
                    break;
                }
            }
            if (!flag) {
                return "图片格式不正确";
            }
            // 判断文件内容
            BufferedImage image = ImageIO.read(file.getInputStream());
            if (image != null) {
                /*System.err.println(String.valueOf(image.getHeight()));
                System.err.println(String.valueOf(image.getWidth()));*/
            } else {
                return "文件内容不正确";
            }

            // 创建OSSClient实例。
            inputStream = file.getInputStream();

            //获取文件名称
            String fileName = file.getOriginalFilename();
            //1 在文件名称里面添加随机唯一的值
            String newName = UUID.randomUUID().toString().replaceAll("-", "") + fileName;
            //2 把文件按照日期进行分类
            String datePath = new DateTime().toString("yyyy/MM/dd");
            String urlPath = ConstantPropertiesUtils.FILE_HOST + "/" + datePath + "/" + newName;

            // 填写Bucket名称和Object完整路径。Object完整路径中不能包含Bucket名称。
            ossClient.putObject(bucketName, urlPath, inputStream);

            return "https://" + bucketName + "." + endpoint + "/" + urlPath;
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        } finally {
            // 关闭OSSClient。
            ossClient.shutdown();
        }

    }
}
